plant system

Code Appendix

                from flask import Flask, render_template, jsonify, Response, request
                from picamera import PiCamera
                import RPi.GPIO as GPIO
                from time import sleep
                import io
                import adafruit_dht
                import board
                import time
                
                app = Flask(__name__)
                # Servo setup
                SERVO_PIN = 17
                LIGHT_PIN = 21
                GPIO.setmode(GPIO.BCM)
                GPIO.setup(SERVO_PIN, GPIO.OUT)
                GPIO.setup(LIGHT_PIN, GPIO.OUT)
                pwm = GPIO.PWM(SERVO_PIN, 50)
                pwm.start(0)
                
                current_angle = 0  # Global variable to track the servo angle
                light_status = False
                plant_humidity_thresholds = {
                    'tomato': 50,
                    'potato': 64,
                    'lettuce': 30
                }
                # Initialize the DHT11 sensor once
                dhtDevice = adafruit_dht.DHT11(board.D4, use_pulseio=False)
                def adjust_servo_based_on_humidity(humidity, threshold):
                    global current_angle
                    if humidity < threshold and current_angle != 90:
                        set_servo_angle(90)
                        current_angle = 90
                    elif humidity >= threshold and current_angle != 0:
                        set_servo_angle(0)
                        current_angle = 0
                
                def get_sensor_data():
                    for _ in range(3):  # Try up to 3 times
                        try:
                            temperature = dhtDevice.temperature
                            humidity = dhtDevice.humidity
                            if temperature is not None and humidity is not None:
                                return temperature, humidity
                        except RuntimeError as e:
                            print("Retrying sensor read:", e)
                            time.sleep(2.0)
                    return None, None  # Indicate failure after retries
                
                def toggle_light_status():
                    global light_status
                    light_status = not light_status
                    GPIO.output(LIGHT_PIN, light_status)
                
                
                def set_servo_angle(angle):
                    duty_cycle = angle / 18 + 2
                    pwm.ChangeDutyCycle(duty_cycle)
                    time.sleep(1)
                    pwm.ChangeDutyCycle(0)
                @app.route('/toggle-light', methods=['POST'])
                def toggle_light():
                    toggle_light_status()
                    return jsonify({"message": "Light toggled"})
                @app.route('/toggle-pump', methods=['POST'])
                def toggle_pump():
                    global current_angle
                    # Toggle between 0 and 90 degrees
                    if current_angle == 0:
                        new_angle = 90
                    else:
                        new_angle = 0
                
                    set_servo_angle(new_angle)
                    current_angle = new_angle
                    return jsonify({"new_angle": current_angle})
                
                def generate_video_stream():
                    with PiCamera() as camera:
                        camera.resolution = (640, 480)
                        camera.framerate = 24
                        stream = io.BytesIO()
                
                        for _ in camera.capture_continuous(stream, 'jpeg', use_video_port=True):
                            stream.seek(0)
                            frame = stream.read()
                            yield (b'--frame\r\n'
                                   b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
                            stream.seek(0)
                            stream.truncate()
                
                @app.route('/')
                def index():
                    return render_template('index.html')
                
                @app.route('/sensor-data')
                def sensor_data():
                    temperature, humidity = get_sensor_data()
                    if temperature is not None and humidity is not None:
                        selected_plant = request.args.get('plant', 'tomato')  # 默认为番茄
                        adjust_servo_based_on_humidity(humidity, plant_humidity_thresholds[selected_plant])
                        return jsonify({'temperature': temperature, 'humidity': humidity})
                    else:
                        return jsonify({'temperature': 'N/A', 'humidity': 'N/A'})
                @app.route('/video-stream')
                def video_stream():
                    return Response(generate_video_stream(),
                                    mimetype='multipart/x-mixed-replace; boundary=frame')
                
                if __name__ == '__main__':
                    app.run(host='0.0.0.0', port=5000, debug=True)